#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# @Project : chat_model 
# @File    : video_model_schema.py
# @IDE     : PyCharm 
# @Author  :ZH
# @Time    : 2025/1/3 16:22
from pydantic import Field, BaseModel


class BaseVideoModelSchema(BaseModel):
    model_name: str = Field(description='模型名称', title='model_name')
    model_base_url: str = Field(description='模型基础地址', title='model_base_url')
    firm: int = Field(description='模型厂商', title='firm')
    api_key: str = Field(description='api_key', title='api_key')
    description: str = Field(description='描述', title='description')
    is_default: bool = Field(description='是否默认', title='is_default')


class CreateVideoModelSchema(BaseModel):
    __doc__ = "创建视频生成模型模式"


class UpdateVideoModelSchema(BaseModel):
    __doc__ = "更新视频生成模型模式"
